package LeetCode;

import java.util.Arrays;

/**
 * Created by kechelle on 2020/8/16 12:33
 * 分发糖果
 * 老师想给孩子们分发糖果，有N个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。
 * 你需要按照以下要求，帮助老师给这些孩子分发糖果：
 *
 * 每个孩子至少分配到 1个糖果。
 * 相邻的孩子中，评分高的孩子必须获得更多的糖果。
 * 那么这样下来，老师至少需要准备多少颗糖果呢？
 *
 */
public class Code135 {
    //方法1
    public static int candy(int[] ratings){
        int[] nums = new int[ratings.length];
        //初始化数组，每个元素填充1
        Arrays.fill(nums,1);
        //从左往右开始分配
        for (int i=1;i< ratings.length;i++){
            if (ratings[i]>ratings[i-1]){
                nums[i] = nums[i-1] + 1;
            }
        }
        int result = nums[ratings.length-1];
        for (int i= ratings.length-2;i>=0;i--){
            if (ratings[i]>ratings[i+1] && nums[i]<=nums[i+1]){
                nums[i] = nums[i+1] + 1;
            }
            result += nums[i];
        }
        System.out.println(Arrays.toString(nums));
        return result;
    }

    //方法2
    public static int candy2(int[] ratings){
        int[] left = new int[ratings.length];
        int[] right = new int[ratings.length];
        Arrays.fill(left,1);
        Arrays.fill(right,1);
        //从左到右开始分发
        for (int i=1;i< ratings.length;i++){
            if (ratings[i]>ratings[i-1]) left[i] = left[i-1] + 1;
        }
        //从右到左开始分发
        for (int i=ratings.length-2;i>=0;i--){
            if (ratings[i]>ratings[i+1]) right[i] = right[i+1] + 1;
        }
        int result = 0;
        for (int i=0;i< ratings.length;i++){
            result = result + Math.max(left[i],right[i]);
        }
        return result;
    }

    public static void main(String[] args) {
        //int[] rantings = {1,0,2};
        int[] rantings = {1,3,4,5,2};
        System.out.println(Code135.candy(rantings));
    }
}
